package com.example.demo.common.config;

import com.example.demo.common.component.WebSocketServer;
import com.example.demo.project.dao.QQUserDao;
import com.example.demo.project.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration
public class WebSocketConfig {
    /**
     * 注入一个ServerEndpointExporter,该Bean会自动注册使用@ServerEndpoint注解申明的websocket endpoint
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
    @Autowired
    public void userService(UserService userService){
        WebSocketServer.userService = userService;
    }
    @Autowired
    public void mongoTemplate(MongoTemplate mongoTemplate){
        WebSocketServer.mongoTemplate = mongoTemplate;
    }
    @Autowired
    public void qqUserDao(QQUserDao qqUserDao){
        WebSocketServer.qqUserDao = qqUserDao;
    }
}
